/*
 * Copyright 2018 peipeihh
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * limitations under the License.
 */

package com.pphh.demo.cluster;

import com.pphh.demo.job.SimpleRecoveryJob;
import org.quartz.*;
import org.quartz.impl.StdSchedulerFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Date;

/**
 * ClusterScheduler
 *
 * @author huangyinhuang
 * @date 6/5/2018
 */
public class ClusterScheduler {

    private static Logger log = LoggerFactory.getLogger(ClusterScheduler.class);

    /**
     * 创建任务调度或者集群执行任务调度
     *
     * @param bSchedulerNewJob True 创建任务调度，False 集群执行任务调度
     * @throws SchedulerException
     * @throws InterruptedException
     */
    public void run(Boolean bSchedulerNewJob) throws SchedulerException, InterruptedException {
        SchedulerFactory factory = new StdSchedulerFactory();
        Scheduler scheduler = factory.getScheduler();

        if (bSchedulerNewJob) {

            log.info("clear any existing job scheduler before creating new one.");
            scheduler.clear();

            JobDetail job = JobBuilder.newJob(SimpleRecoveryJob.class)
                    .withIdentity("MyJob", "MyGroup")
                    .requestRecovery()
                    .build();

            CronTrigger trigger = TriggerBuilder.newTrigger()
                    .withIdentity("MyTrigger", "MyGroup")
                    .startAt(DateBuilder.evenMinuteDate(new Date()))
                    .withSchedule(CronScheduleBuilder.cronSchedule("0/10 * * * * ?"))
                    .build();

            log.info("start new job scheduler.");
            scheduler.scheduleJob(job, trigger);

            log.info("start the scheduler.");
            scheduler.start();

            log.info("shutdown the scheduler gracefully.");
            scheduler.shutdown(true);
            log.info("the end.");

        } else {

            log.info("start the scheduler.");
            scheduler.start();

            log.info("sleeping for scheduler to run job.");
            Thread.sleep(600L * 1000L);

            log.info("shutdown the scheduler gracefully.");
            scheduler.shutdown(true);
            log.info("the end.");

        }

    }
}
